from pydantic import BaseModel


class RagInput(BaseModel):
    api_key: str
    user_input: str